import jieba
import wordcloud
from imageio.v3 import imread
txt = open("text/threekingdoms.txt", "r", encoding="utf-8").read()
excludes = {"将军","却说","荆州","二人","不可","不能","如此","商议","如何","左右"}
words = jieba.lcut(txt)
counts = {}
for word in words:
    if len(word) == 1:
        continue
    elif word == "诸葛亮" or word == "孔明曰":
        rword = "孔明"
    elif word == "关公" or word == "云长":
        rword = "关羽"
    elif word == "玄德" or word == "玄德曰":
        rword = "刘备"
    elif word == "孟得" or word == "丞相":
        rword = "曹操"
    else:
        rword = word
    counts[rword] = counts.get(rword,0)+1
for word in excludes:
    del counts[word]
ls = " ".join(words)
items = list(counts.items())
items.sort(key=lambda x:x[1], reverse=True)
for i in range(15):
    word, count = items[i]
    print("{0:<10}{1:>5}".format(word,count))

wordscloud = wordcloud.WordCloud(stopwords={"之","我","也","为","乃","曰","将","又","与","吾","来","皆","有","见","人","矣","在","了","于"}, max_font_size=100, min_font_size=15, max_words=150, font_path="font/SongTi.ttf", width=1000, height=700, background_color="white", mask=imread(
    "mark/photo2.png"))
wordscloud.generate(ls)
wordscloud.to_file("cloud_to_three.png")